


from rest_framework.generics import ListCreateAPIView

############用户管理#############
############获取用户,保存用户##################
from apps.meiduo_admin.serializers.users import UserSerializer
from apps.meiduo_admin.utils import PageNum
from apps.users.models import User


class UserView(ListCreateAPIView):
    '''
    list:获取用户
    create:保存用户
    '''
    # 指定序列化器
    serializer_class = UserSerializer
    # 指定查询集
    queryset = User.objects.filter(is_staff=False)
    #指定分页器
    pagination_class = PageNum

    # 重写get_queryset
    def get_queryset(self):
        # 根据keyword参数返回不同查询结果
        keyword = self.request.query_params.get("keyword")

        if keyword == "" or keyword is None:
            # 返回所有数据
            return User.objects.filter(is_staff=False)

        else:
            # 返回查询数据
            return User.objects.filter(is_staff=False,username__contains=keyword)